package com.javaclimb.book.controller;

import com.javaclimb.book.common.ResultMapUtil;
import org.apache.shiro.SecurityUtils;
import org.apache.shiro.authc.IncorrectCredentialsException;
import org.apache.shiro.authc.UnknownAccountException;
import org.apache.shiro.authc.UsernamePasswordToken;
import org.apache.shiro.subject.Subject;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.ResponseBody;

import java.util.HashMap;

/**
 * 用户相关的controller
 */
@Controller
public class UserController {

    /**
     * 转向登录页面
     */
    @RequestMapping(value = "/login")
    public String login() {
        return "/login";
    }
    /**
     * 转向管理员登录页面
     */
    @RequestMapping(value = "/adminLogin")
    public String adminLogin() {
        return "/adminLogin";
    }

    @RequestMapping(value = "/favicon.ico")
    @ResponseBody
    public HashMap favicon() {
        HashMap<Object, Object> objectObjectHashMap = new HashMap<>();
        objectObjectHashMap.put("icon", "123");
        return objectObjectHashMap;
    }

    @RequestMapping(value = "/")
    public String dashmain() {
        return "redirect:/login";
    }

    /**
     * 判断用户登录是否成功
     */
    @RequestMapping(value = "/toLogin")
    @ResponseBody
    public Object toLogin(String username, String password) {
        if (username == null || password == null) {
            return ResultMapUtil.getHashMapLogin("用户名密码不能为空", "2");
        }
        Subject subject = SecurityUtils.getSubject();
        UsernamePasswordToken token = new UsernamePasswordToken(username, password);
        try {
            subject.login(token);
        } catch (UnknownAccountException e) {
            return ResultMapUtil.getHashMapLogin("用户名不存在", "2");
        } catch (IncorrectCredentialsException e) {
            return ResultMapUtil.getHashMapLogin("密码错误", "2");
        }
        return ResultMapUtil.getHashMapLogin("验证成功", "1");
    }

    /**
     * 转向后台管理首页
     */
    @RequestMapping(value = "/index")
    public String index() {
        return "/index";
    }

    /**
     * 退出登录
     */
    @RequestMapping(value = "/logout")
    public String logout() {
        Subject subject = SecurityUtils.getSubject();
        subject.logout();
        return "redirect:/login";
    }

}
